package com.idlecode.util.sm;

import org.bouncycastle.crypto.Digest;
import org.bouncycastle.crypto.digests.SM3Digest;
import org.bouncycastle.util.encoders.Base64;
import org.bouncycastle.util.encoders.Hex;

import java.io.File;
import java.io.FileInputStream;
import java.io.IOException;

/**
 * @program: IdleCode
 * @description
 * @author: rongjuwei
 * @create: 2024-10-25 10:04
 **/
public class SM3Util {

    /**
     * 生成字符串的SM3哈希值（Hex格式）
     * @param inputData 待哈希的输入数据
     * @return SM3哈希后的十六进制字符串
     */
    public static String hashToHex(String inputData) {
        byte[] hashBytes = calculateHash(inputData.getBytes());
        return Hex.toHexString(hashBytes);
    }

    /**
     * 生成字符串的SM3哈希值（Base64格式）
     * @param inputData 待哈希的输入数据
     * @return SM3哈希后的Base64字符串
     */
    public static String hashToBase64(String inputData) {
        byte[] hashBytes = calculateHash(inputData.getBytes());
        return Base64.toBase64String(hashBytes);
    }

    /**
     * 生成文件的SM3哈希值（Hex格式）
     * @param file 待哈希的文件
     * @return SM3哈希后的十六进制字符串
     * @throws IOException 文件读取异常
     */
    public static String hashFileToHex(File file) throws IOException {
        byte[] hashBytes = calculateFileHash(file);
        return Hex.toHexString(hashBytes);
    }

    /**
     * 生成文件的SM3哈希值（Base64格式）
     * @param file 待哈希的文件
     * @return SM3哈希后的Base64字符串
     * @throws IOException 文件读取异常
     */
    public static String hashFileToBase64(File file) throws IOException {
        byte[] hashBytes = calculateFileHash(file);
        return Base64.toBase64String(hashBytes);
    }

    /**
     * 计算字节数组的SM3哈希值
     * @param inputDataBytes 输入数据字节数组
     * @return 哈希值的字节数组
     */
    private static byte[] calculateHash(byte[] inputDataBytes) {
        Digest digest = new SM3Digest();
        digest.update(inputDataBytes, 0, inputDataBytes.length);
        byte[] hashBytes = new byte[digest.getDigestSize()];
        digest.doFinal(hashBytes, 0);
        return hashBytes;
    }

    /**
     * 计算文件的SM3哈希值
     * @param file 文件
     * @return 哈希值的字节数组
     * @throws IOException 文件读取异常
     */
    private static byte[] calculateFileHash(File file) throws IOException {
        Digest digest = new SM3Digest();
        FileInputStream fis = null;

        try {
            fis = new FileInputStream(file);
            byte[] buffer = new byte[4096]; // 每次读取4KB
            int bytesRead;

            while ((bytesRead = fis.read(buffer)) != -1) {
                digest.update(buffer, 0, bytesRead);
            }

            byte[] hashBytes = new byte[digest.getDigestSize()];
            digest.doFinal(hashBytes, 0);
            return hashBytes;
        } finally {
            if (fis != null) {
                fis.close(); // 关闭文件输入流
            }
        }
    }

    public static void main(String[] args) {
        try {
            String data = "Hello World";

            // 输出字符串哈希值（Hex格式）
            String hexHash = SM3Util.hashToHex(data);
            System.out.println("SM3 Hash (Hex): " + hexHash);

            // 输出字符串哈希值（Base64格式）
            String base64Hash = SM3Util.hashToBase64(data);
            System.out.println("SM3 Hash (Base64): " + base64Hash);

            // 对文件进行哈希计算（Hex和Base64格式）
            File file = new File("path/to/your/file.txt");

            String fileHexHash = SM3Util.hashFileToHex(file);
            System.out.println("SM3 File Hash (Hex): " + fileHexHash);

            String fileBase64Hash = SM3Util.hashFileToBase64(file);
            System.out.println("SM3 File Hash (Base64): " + fileBase64Hash);

        } catch (IOException e) {
            e.printStackTrace();
        }
    }
}
